python improving performance